// Copyright (c) 2024, BlockProject 3D

//

// All rights reserved.

//

// Redistribution and use in source and binary forms, with or without modification,

// are permitted provided that the following conditions are met:

//

//     * Redistributions of source code must retain the above copyright notice,

//       this list of conditions and the following disclaimer.

//     * Redistributions in binary form must reproduce the above copyright notice,

//       this list of conditions and the following disclaimer in the documentation

//       and/or other materials provided with the distribution.

//     * Neither the name of BlockProject 3D nor the names of its contributors

//       may be used to endorse or promote products derived from this software

//       without specific prior written permission.

//

// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR

// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR

// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF

// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING

// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS

// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

use bp3d_protoc::api::core::generator::Params;

use clap::{Parser, ValueEnum};

use std::ffi::OsString;

use std::path::{Path, PathBuf};

#[derive(ValueEnum, Copy, Clone, Debug)]

pub enum Generator {

    /// The Rust code generator.

    Rust,

    /// The Swift code generator.

    Swift,

}

#[derive(ValueEnum, Copy, Clone, Debug)]

pub enum Feature {

    /// The generated code should be able to write messages.

    WriteMessages,

    /// The generated code should be able to read messages.

    ReadMessages,

    /// The generated code needs enums.

    UseEnums,

    /// The generated code needs structures.

    UseStructs,

    /// The generated code needs messages.

    UseMessages,

    /// The generated code needs unions.

    UseUnions,

}

impl Feature {

    pub fn apply(&self, params: &mut Params) {

        match self {

            Feature::WriteMessages => params.write_messages = true,

            Feature::ReadMessages => params.read_messages = true,

            Feature::UseEnums => params.use_enums = true,

            Feature::UseStructs => params.use_structs = true,

            Feature::UseMessages => params.use_messages = true,

            Feature::UseUnions => params.use_unions = true,

        };

    }

}

#[derive(Parser, Debug)]

#[command(version, about, long_about = None)]

/// CLI tool to call BP3D protocol compiler outside build-scripts and access to source code.

pub struct Args {

    /// List of imported protocols, each by pairs of path to the protocol description and import

    /// path.

    #[clap(short = 'i', long = "import", number_of_values = 2)]

    pub imports: Vec<OsString>,

    /// List of input protocol description files to compile.

    #[clap(required=true, num_args=1..)]

    pub inputs: Vec<PathBuf>,

    /// Output directory where to place the generated protocols (an additional directory is created

    /// for each protocol to be compiled).

    #[clap(short = 'o', long = "output")]

    pub output: Option<PathBuf>,

    /// Name of the code generator to use.

    #[clap(short = 'g', long = "generator", default_value = "rust")]

    pub generator: Generator,

    /// Features to enable, the default is to use the default set of features which includes

    /// everything except reading and writing messages.

    #[clap(short = 'f', long = "feature")]

    pub features: Option<Vec<Feature>>,

    /// The file header to include at the top of each generated file, each line in the file header

    /// is already formatted according to the line comments syntax of the chosen target generation

    /// language.

    #[clap(long = "header")]

    pub file_header: Option<PathBuf>,

}

impl Args {

    pub fn iter_imports(&self) -> impl Iterator<Item = (&Path, &str)> {

        self.imports.chunks(2).map(|v| (Path::new(&v[0]), v[1].to_str().unwrap()))

    }

}